﻿using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Collections.Generic;
using System.Data;
using System.IO;

namespace WpfSnqkGasAnalysis.Model
{


    /// <summary>
    /// CSV文件转换类
    /// </summary>
    /// 创建时间：2023-1-3 14:44:49
    public static class CsvHelper
    {
        /// <summary>
        /// 导出报表为Csv
        /// </summary>
        /// <param name="rows">导出的行数据</param>
        /// <param name="strFilePath">物理路径</param>
        /// <param name="tableHeader">表头</param>
        /// <param name="columName">字段标题,逗号分隔</param>
        public static void ExportMakeCsv(List<List<object>> rows, string strFilePath, string tableHeader, string columName)
        {
            string dir = Path.GetDirectoryName(strFilePath);
            if (!Directory.Exists(dir))
            {
                Directory.CreateDirectory(dir);
            }

            StreamWriter strmWriterObj = new StreamWriter(strFilePath, false, System.Text.Encoding.UTF8);
            strmWriterObj.WriteLine(tableHeader);//表头
            strmWriterObj.WriteLine(columName);//列标题
            foreach (var cellValues in rows)
            {
                string txt = string.Join(",", cellValues);

                //替换英文逗号为分号。有些单元格数据包含逗号，不处理会占用右侧单元格
                if (cellValues.Exists(x => x != null && x.ToString().Contains(",")))
                {
                    List<object> douhaoArr = new List<object>();
                    for (int i = 0; i < cellValues.Count; i++)
                    {
                        var g = cellValues[i];
                        if (g != null && g.ToString().Contains(","))
                        {
                            g = g.ToString().Replace(",", ";");
                        }
                        douhaoArr.Add(g);
                    }
                    txt = string.Join(",", douhaoArr);
                }
                strmWriterObj.WriteLine(txt);//数据行
            }
            strmWriterObj.Close();
            strmWriterObj.Dispose();
        }

        /// <summary>
        /// 导出报表为Csv
        /// </summary>
        /// <param name="dt">DataTable</param>
        /// <param name="strFilePath">物理路径</param>
        /// <param name="tableheader">表头</param>
        /// <param name="columname">字段标题,逗号分隔</param>
        public static bool DataTableMakeCsv(DataTable dt, string strFilePath, string tableHeader, string columName)
        {
            try
            {
                string dir = Path.GetDirectoryName(strFilePath);
                if (!Directory.Exists(dir))
                {
                    Directory.CreateDirectory(dir);
                }
                string strBufferLine = "";
                StreamWriter strmWriterObj = new StreamWriter(strFilePath, false, System.Text.Encoding.UTF8);
                strmWriterObj.WriteLine(tableHeader);
                strmWriterObj.WriteLine(columName);
                for (int i = 0; i < dt.Rows.Count; i++)
                {
                    strBufferLine = "";
                    for (int j = 0; j < dt.Columns.Count; j++)
                    {
                        if (j > 0)
                            strBufferLine += ",";
                        strBufferLine += dt.Rows[i][j].ToString();
                    }
                    strmWriterObj.WriteLine(strBufferLine);
                }
                strmWriterObj.Close();
                return true;
            }
            catch
            {
                return false;
            }
        }

        /// <summary>
        /// 将Csv读入DataTable
        /// </summary>
        /// <param name="filePath">csv文件路径</param>
        /// <param name="n">表示第n行是字段title,第n+1行是记录开始</param>
        public static DataTable ReadCsv2DataTable(string filePath, int n, DataTable dt)
        {
            string dir = Path.GetDirectoryName(filePath);
            if (!Directory.Exists(dir))
            {
                Directory.CreateDirectory(dir);
            }

            StreamReader reader = new StreamReader(filePath, System.Text.Encoding.UTF8, false);
            int i = 0, m = 0;
            reader.Peek();
            while (reader.Peek() > 0)
            {
                m = m + 1;
                string str = reader.ReadLine();
                if (str == null) continue;
                if (m >= n + 1)
                {
                    string[] split = str.Split(',');

                    System.Data.DataRow dr = dt.NewRow();
                    for (i = 0; i < split.Length; i++)
                    {
                        dr[i] = split[i];
                    }
                    dt.Rows.Add(dr);
                }
            }
            return dt;
        }
    }



}
